Rainfall-Runoff Prediction at Multiple Timescales with a Single Long Short-Term Memory Network

New LSTM-based architecture for predictions at multiple temporal time scales.

Abstract

Long Short-Term Memory Networks (LSTMs) have been applied to daily discharge prediction with remarkable success. Many practical scenarios, however, require predictions at more granular timescales. For instance, accurate prediction of short but extreme flood peaks can make a life-saving difference, yet such peaks may escape the coarse temporal resolution of daily predictions. Naively training an LSTM on hourly data, however, entails very long input sequences that make learning hard and computationally expensive. In this study, we propose two Multi-Timescale LSTM (MTS-LSTM) architectures that jointly predict multiple timescales within one model, as they process long-past inputs at a single temporal resolution and branch out into each individual timescale for more recent input steps. We test these models on 516 basins across the continental United States and benchmark against the US National Water Model. Compared to naive prediction with a distinct LSTM per timescale, the multi-timescale architectures are computationally more efficient with no loss in accuracy. Beyond prediction quality, the multi-timescale LSTM can process different input variables at different timescales, which is especially relevant to operational applications where the lead time of meteorological forcings depends on their temporal resolution.

Paper

Gauch, M., Kratzert, F., Klotz, D., Nearing, G., Lin, J., and Hochreiter, S.: Rainfall–runoff prediction at multiple timescales with a single Long Short-Term Memory network, Hydrol. Earth Syst. Sci., 25, 2045–2062, 2021.

Code

The results of this paper were produced with the neuralHydrology Python package. All details to reproduce the results of this paper can be found at https://github.com/gauchm/mts-lstm.

Citation

@article{Gauch2021mtslstm,
    author = {Gauch, M. and Kratzert, F. and Klotz, D. and Nearing, G. and Lin, J. and Hochreiter, S.},
    title = {Rainfall--runoff prediction at multiple timescales with a single Long Short-Term Memory network},
    journal = {Hydrology and Earth System Sciences},
    volume = {25},
    year = {2021},
    number = {4},
    pages = {2045--2062},
    url = {https://hess.copernicus.org/articles/25/2045/2021/},
    doi = {10.5194/hess-25-2045-2021}
}